package cn.seecoder.ai.algorithm.ml

import cn.seecoder.ai.model.bo.TrainParamsBO
import org.apache.spark.ml.PipelineModel
import org.apache.spark.sql.SparkSession

import java.util

trait BaseAlgorithm {


  def train(sparkSession: SparkSession,
            libSvmFileUri: String,
            trainParams: TrainParamsBO): (PipelineModel, util.Map[String, Object])

}
